package com.fz.utils;

import com.alibaba.excel.EasyExcel;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/***
 *  导出文件  通过对应的vo在字段上加入 @ExcelProperty 注解 进行导出
 *  不需要导出的字段加入@ExcelIgnore注解
 */

public class ExcelUtils {

    /**
     * @方法描述: 导出文件
     * @参数: list代表需要导出的数据，tClass代表 对应的实体vo，fileName 文件名
     * @返回值:
     * @创建人: I am God
     * @创建时间:: 2021/3/29
     */
    public static<T> void exportExcel(List<T> list,Class<T> tClass, String fileName, HttpServletResponse response) throws IOException {
        try {
            response.setContentType("application/vnd.ms-excel");
            response.setCharacterEncoding("utf-8");
            // 这里URLEncoder.encode可以防止中文乱码
            String name = URLEncoder.encode(fileName, "UTF-8");
            response.setHeader("Content-disposition", "attachment;filename=" + name + ".xlsx");
            // 这里需要设置不关闭流
            EasyExcel.write(response.getOutputStream(), tClass).autoCloseStream(Boolean.FALSE).sheet("sheet ")
                    .doWrite(list);
        } catch (Exception e) {
            // 重置response
            response.reset();
            response.setContentType("application/json");
            response.setCharacterEncoding("utf-8");
            Map<String, String> map = new HashMap<String, String>();
            map.put("status", "failure");
            map.put("message", "下载文件失败" + e.getMessage());
            response.getWriter().println(JacksonUtils.obj2String(map));
        }
    }
}
